# +----------------------------------------------------------------------
# | ChatWork智能聊天办公系统
# +----------------------------------------------------------------------
# | 软件声明: 本系统并非自由软件,未经授权任何形式的商业使用均属非法。
# | 版权保护: 任何企业和个人不允许对程序代码以任何形式任何目的复制/分发。
# | 授权要求: 如有商业使用需求,请务必先与版权所有者取得联系并获得正式授权。
# +----------------------------------------------------------------------
# | Author: ChatWork Team <2474369941@qq.com>
# +----------------------------------------------------------------------
class SubscribeEnum:
    # 应用定义
    APPLY_CHAT = "chat"                  # 对话模型
    APPLY_VECTOR = "vector"              # 向量模型
    APPLY_RANKING = "ranking"            # 重排模型
    APPLY_DRAWS_MJ = "draws_mj"          # AI绘画(MJ)
    # APPLY_DRAWS_SD = "draws_sd"          # AI绘画(SD)
    APPLY_DRAWS_DALLE = "draws_dalle"    # AU绘画(DALLE)
    APPLY_DRAWS_DOUBAO = "draws_doubao"  # AI绘画(豆包)
    APPLY_MUSIC_SUNO = "music_suno"      # AI音乐(SUNO)
    APPLY_VIDEO_LUMA = "video_luma"      # AI视频(LUMA)
    APPLY_VIDEO_ZHIPU = "video_zhipu"    # AI视频(智谱)

    @classmethod
    def get_apply_lists(cls, code: str = None):
        _desc = {
            cls.APPLY_DRAWS_MJ: "AI绘画(MJ)",
            # cls.APPLY_DRAWS_SD: "AI绘画(SD)",
            cls.APPLY_DRAWS_DALLE: "AI绘画(DALLE)",
            cls.APPLY_DRAWS_DOUBAO: "AI绘画(豆包)",
            cls.APPLY_MUSIC_SUNO: "AI音乐(SUNO)",
            cls.APPLY_VIDEO_LUMA: "AI视频(LUMA)",
            cls.APPLY_VIDEO_ZHIPU: "AI视频(智谱)",
        }
        if code is None:
            return _desc
        return _desc.get(code, "")
